package com.leetCode.ArrayDemo;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class ThreeSumClosestDemo {

    public static void main(String[] args) {
        int[] nums1 = new int[]{0,1,2};
        int[] nums2 = new int[]{-4,-1,1,2};
        int[] nums3 = new int[]{-5,-5,-4,0,0,3,3,4,5};


        int[] nums = new int[]{4,0,5,-5,3,3,0,-4,-5};
        System.out.println(threeSumClosest(nums,-2));
    }
    /**
     * 给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数，使它们的和与 target 最接近。
     *
     * 返回这三个数的和。
     *
     * 假定每组输入只存在恰好一个解。
     *
     * 示例 1：
     *
     * 输入：nums = [-1,2,1,-4], target = 1
     * 输出：2
     * 解释：与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
     * 示例 2：
     *
     * 输入：nums = [0,0,0], target = 1
     * 输出：0
     * 提示：
     *
     * 3 <= nums.length <= 1000
     * -1000 <= nums[i] <= 1000
     * -104 <= target <= 104
     * Related Topics
     * 数组
     * 双指针
     * 排序
     */
    public static int threeSumClosest(int[] nums, int target) {
        Arrays.sort(nums);
        int n = nums.length;
        int best = 10000000;

        // 枚举 a
        for (int i = 0; i < n; ++i) {
            // 保证和上一次枚举的元素不相等
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            // 使用双指针枚举 b 和 c
            int j = i + 1, k = n - 1;
            while (j < k) {
                int sum = nums[i] + nums[j] + nums[k];
                // 如果和为 target 直接返回答案
                if (sum == target) {
                    return target;
                }
                // 根据差值的绝对值来更新答案
                if (Math.abs(sum - target) < Math.abs(best - target)) {
                    best = sum;
                }
                if (sum > target) {
                    // 如果和大于 target，移动 c 对应的指针
                    int k0 = k - 1;
                    // 移动到下一个不相等的元素
                    while (j < k0 && nums[k0] == nums[k]) {
                        --k0;
                    }
                    k = k0;
                } else {
                    // 如果和小于 target，移动 b 对应的指针
                    int j0 = j + 1;
                    // 移动到下一个不相等的元素
                    while (j0 < k && nums[j0] == nums[j]) {
                        ++j0;
                    }
                    j = j0;
                }
            }
        }
        return best;
    }

    /**
     * 排序加双指针
     */
    public static int threeSumClosest1(int[] nums, int target) {
        Arrays.sort(nums);
        int n = nums.length;
        int result = 10000000;
        for (int i = 0;i < n;++i){
            if (i > 0 && nums[i] == nums[i - 1]){
                continue;
            }
            //枚举其余两个指针
            int l = i + 1,r = n - 1;
            while (l < r){
                int sum = nums[i] + nums[l] + nums[r];
                if (sum == target) {
                    return target;
                }
                //根据差值绝对值更新结果
                if (Math.abs(sum - target) < Math.abs(result - sum)){
                    result = sum;
                }
                if (sum > target){
                    //如果sum > target,那么向左移动r指针
                    int r0 = r - 1;
                    while (r0 >= l && nums[r0] == nums[r]){
                        --r0;
                    }
                    r = r0;
                } else {
                   //如果sum < target,那么向右移动l指针
                    int l0 = l + 1;
                    while (l0 < r && nums[l0] == nums[l]){
                        ++l0;
                    }
                    l = l0;
                }
            }
        }
        return result;
    }
}
